How can I select functions conditionally instead of randomly using numpy in Python